/*
 * Copyright © 2018 www.noark.xyz All Rights Reserved.
 *
 * 感谢您选择Noark框架，希望我们的努力能为您提供一个简单、易用、稳定的服务器端框架 ！
 * 除非符合Noark许可协议，否则不得使用该文件，您可以下载许可协议文件：
 *
 *        http://www.noark.xyz/LICENSE
 *
 * 1.未经许可，任何公司及个人不得以任何方式或理由对本框架进行修改、使用和传播;
 * 2.禁止在本项目或任何子项目的基础上发展任何派生版本、修改版本或第三方版本;
 * 3.无论你对源代码做出任何修改和改进，版权都归Noark研发团队所有，我们保留所有权利;
 * 4.凡侵犯Noark版权等知识产权的，必依法追究其法律责任，特此郑重法律声明！
 */
package xyz.noark.network.rpc.shuttle;

import io.netty.channel.Channel;
import xyz.noark.network.codec.AbstractPacketCodec;
import xyz.noark.network.codec.shuttle.ShuttlePacketCodec;
import xyz.noark.network.rpc.serializer.RpcJsonSerializer;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import static xyz.noark.log.LogHelper.logger;

/**
 * 连接管理类，主要维护所有连接关系，不区分客户端还是服务器
 *
 * @author 小流氓[176543888@qq.com]
 * @since 4.0
 */
public class ShuttleConnectionManager {
    // 存储所有连接: UID -> ShuttleConnection
    private final ConcurrentMap<Integer, ShuttleConnection> connectionMap = new ConcurrentHashMap<>();
    private final ShuttleConnectionFactory connectionFactory;
    private final AbstractPacketCodec packetCodec;

    public ShuttleConnectionManager() {
        this(new ShuttlePacketCodec(new RpcJsonSerializer()));
    }

    public ShuttleConnectionManager(AbstractPacketCodec packetCodec) {
        this.packetCodec = packetCodec;
        this.connectionFactory = new ShuttleConnectionFactory(this, packetCodec);
    }

    /**
     * 注册一个新连接（主动或被动）
     */
    public void register(Integer serverUid, Channel channel) {
        connectionMap.put(serverUid, new ShuttleConnection(serverUid, channel, packetCodec));
        logger.info("shuttle connection register. uid={}, address={}", serverUid, channel.remoteAddress());
    }

    /**
     * 移除连接, 要把连接也处理掉
     */
    public void unregister(Integer serverUid) {
        ShuttleConnection conn = connectionMap.remove(serverUid);
        if (conn != null) {
            connectionFactory.disconnect(serverUid);
        }
    }

    /**
     * 主动连接到目标服务器，并绑定一个业务 UID * * @param serverUid 业务唯一标识（如 client-001, device-A1） * @param host 目标主机 * @param port 目标端口
     */
    public void connectTo(Integer serverUid, String host, int port) {
        synchronized (this) {
            // 检查 UID 是否已存在
            if (connectionMap.containsKey(serverUid)) {
                System.out.println("🔁 UID 已存在，跳过重复连接: " + serverUid);
                return;
            }
            // 委托给工厂发起连接
            connectionFactory.connect(serverUid, host, port);
        }
    }

    /**
     * 根据连接 ID 获取连接
     */
    public ShuttleConnection getConnection(Integer serverUid) {
        return connectionMap.get(serverUid);
    }
}